from wsgiref.validate import check_input

contacts = {
    1:{"姓名":"小明","电话":"001","地址":"广州"},
    2:{"姓名":"小红","电话":"002","地址":"深圳"},
    3:{"姓名":"小王","电话":"003","地址":"北京"},
}

def add_friend():
    id = len(contacts)
    name = input("请输入联系人姓名：").strip()
    phone = input("请输入联系人电话：").strip()
    address= input("请输入联系人地址：").strip()
    contacts[id] = {"姓名":name,"电话":phone,"地址":address}
    print("联系人添加成功！")

def delete_friend():
    id = int(input("请输入要删除的联系人序号(id):"))
    if id in contacts:
        del contacts[id];
        print("删除成功！")
    else:
        print("不存在此联系人！")

def update_friend():
    id = int(input("请输入要修改的联系人id:"))
    if id in contacts:
        name = input("请输入新的联系人姓名：")
        phone = input("请输入新的联系人电话：")
        address = input("请输入新的联系人地址：")
        contacts[id]["姓名"] = name
        contacts[id]["电话"] = phone
        contacts[id]["地址"] = address
        print("联系人信息修改成功！")
    else:
        print("不存在此联系人！")

def get_friend():
    id = int(input("请输入要查找的联系人id:"))
    info = contacts[id]
    if id in contacts:
        print(f"联系人姓名： {info['姓名']}, 电话：{info['电话']}，地址：{info['地址']}")
    else:
        print("不能存在此联系人！")

def menu():
    print("---联系人管理系统---")
    print("---1.添加联系人----")
    print("---2.删除联系人----")
    print("---3.修改联系人----")
    print("---4.查找联系人----")
    print("---5.退出系统----")

def main():
    while True:
        menu()
        choice = input("请选择操作(0-5):")

        if choice == '1':
            add_friend()
        elif choice == '2':
            delete_friend()
        elif choice == '3':
            update_friend()
        elif choice == '4':
            get_friend()
        elif choice == '5':
            print("感谢使用，再见！")
            break
        else:
            print("输入的操作数无效，请重新输入：")

if __name__ == "__main__":
    main()

